nll = nn.ClassNLLCriterion()